cd ../..

gpus=$1
psd_lo_thd=0.10
psd_hi_thd=0.90
weight=0
for bsz in 24 #  
do
    python -m trainer.ATSTContrastiveTrainer fit \
        --config configs/model/ATSTSED_contrastive.yaml \
        --config configs/dataset/desed_with_id.yaml \
        --config configs/trainer/default.yaml \
        --config configs/experiment/intplcontrastive.yaml \
        --config configs/experiment/plugin_ATSTFrame.yaml \
        --model.exp_name="5_cl_usage/ATSTSED/stage1/after_align/+wCL_${weight}/baseline/unlabeled_bsz${bsz}" \
        --data.init_args.batch_sizes=[6,6,12,${bsz}] \
        --model.cl_decay=True \
        --model.cl_weight=${weight} \
        --trainer.devices=${gpus} \
        --trainer.strategy="ddp_find_unused_parameters_false" \
        --model.psd_label_filter=False \
        --model.psd_lo_thd=${psd_lo_thd} \
        --model.psd_hi_thd=${psd_hi_thd} \
        --model.sed_model.init_args.atst_train=False \
        --model.unsupervised_loss.init_args.weight=2 \
        --model.val_psds=False \
        --trainer.check_val_every_n_epoch=10 \
        --model.contrastive_weight.init_args.min_weight=0.5 \
        --data.init_args.contrastive_aug.init_args.mask_ratio=0.4 
done